package org.usmile.algorithms.leetcode.simple;

import java.util.HashMap;
import java.util.Map;

/**
 * 290. 单词规律
 *
 * 给定一种规律 pattern 和一个字符串 s ，判断 s 是否遵循相同的规律。
 * 这里的 遵循 指完全匹配，例如， pattern 里的每个字母和字符串 s 中的每个非空单词之间存在着双向连接的对应规律。
 *
 * 示例1:
 * 输入: pattern = "abba", s = "dog cat cat dog"
 * 输出: true
 *
 * 示例 2:
 * 输入:pattern = "abba", s = "dog cat cat fish"
 * 输出: false
 *
 * 示例 3:
 * 输入: pattern = "aaaa", s = "dog cat cat dog"
 * 输出: false
 *
 * 提示:
 * 1 <= pattern.length <= 300
 * pattern 只包含小写英文字母
 * 1 <= s.length <= 3000
 * s 只包含小写英文字母和 ' '
 * s 不包含 任何前导或尾随对空格
 * s 中每个单词都被 单个空格 分隔
 */
public class _0290 {
    public static void main(String[] args) {
        System.out.println(new _0290_Solution().wordPattern("abba", "dog cat cat dog"));
    }
}

class _0290_Solution {
    public boolean wordPattern(String pattern, String s) {
        String[] words = s.split(" ");
        if (pattern.length() != words.length) {
            return false;
        }

        Map<Character, String> partternMap = new HashMap<>();
        Map<String, Character> wordsMap = new HashMap<>();
        for (int i = 0; i < pattern.length(); i++) {
            char x = pattern.charAt(i);
            String y = words[i];

            if ((partternMap.containsKey(x) && !partternMap.get(x).equals(y))
                    || (wordsMap.containsKey(y) && wordsMap.get(y) != x)) {
                return false;
            }

            partternMap.put(x, y);
            wordsMap.put(y, x);
        }

        return true;
    }
}
